期刊
  出版年
  关键词
结果中检索 Open Search
Please wait a minute...
选择: 显示/隐藏图片
1. 面向牵引座焊缝表面质量检测的轻量型深度学习算法
黄子杰, 欧阳, 江德港, 郭彩玲, 李柏林
《计算机应用》唯一官方网站    2024, 44 (3): 983-988.   DOI: 10.11772/j.issn.1001-9081.2023030349
摘要130)   HTML7)    PDF (3404KB)(89)    收藏

针对人工和传统自动化算法检测牵引座焊缝表面存在检测精度低、速度低的问题,提出一种轻量型的牵引座焊缝表面质量检测算法YOLOv5s-G2CW。首先,用GhostBottleneckV2模块替换YOLOv5s中的C3模块以降低模型的参数量;其次,在YOLOv5s模型的Neck部分引入CBAM(Convolutional Block Attention Module),在通道和空间两个维度上融合焊缝特征;然后将YOLOv5s的定位损失函数改进为Wise-IoU以聚焦普通质量锚框的预测回归;最后移除YOLOv5s模型中用于大物体检测的 13 × 13 特征层以进一步降低模型的参数量。实验结果表明,与YOLOv5s模型相比,YOLOv5s-G2CW的模型大小减小了53.9%,帧率提高了8.0%,平均精度均值(mAP)提高了0.8个百分点,能够满足牵引座焊缝表面质量检测的准确性和实时性要求。

图表 | 参考文献 | 相关文章 | 多维度评价